import erniebot


# 初始化 Erniebot
def initialize_erniebot():
    erniebot.api_type = 'aistudio'
    erniebot.access_token = '721efe09951433260c4ab297c2830cffc82e2d77'


def constitute(user_query):
    initialize_erniebot()

    question = f"""
    帮我根据给的文本续写一段文字。
    最后只输出续写的文字。
    并且续写的 内容不超过30个字。
    
    这是我要续写的文本：{user_query}
    """

    response = erniebot.ChatCompletion.create(
        model='ernie-turbo',
        messages=[
            {'role': 'user', 'content': question},
        ],
        temperature=0.9
    )

    return response.get_result()

# 调用方式示例（在其他文件中，首先要导入这个函数）
# user_question = input()
# result = constitute(user_question)
# print(result)
